3d graph plotting in python